import logging

from flask import g
from flask import render_template, current_app, jsonify
from flask import request
from info.utils.common import user_login_data

from info import constants
from info.models import User, Category, News
from info.utils.response_code import RET
from . import index_blue


# 由于新闻列表的显示只需要局部刷新而不需要渲染模板，所以另起一个路由
@index_blue.route('/news_list')
def get_new_list():
    # 一. 获取数据(可选，若不传值则应用默认值)
    cid = request.args.get('cid', 1)
    page = request.args.get('page', 1)
    per_page = request.args.get('per_page', constants.HOME_PAGE_MAX_NEWS)

    # 二. 参数校验（对类型进行校验）
    try:
        cid = int(cid)
        page = int(page)
        per_page = int(per_page)
    except Exception as e:
        logging.error(e)
        return jsonify(errno=RET.PARAMERR, errmsg='参数有误')

    # 三. 逻辑处理
    # 对新闻进行分类，发布时间降序和页面显示条数进行限制
    # 使用以下方法，可以处理可选参数作为条件的查询
    filters = []
    if cid != 1:
        filters.append(News.category_id == cid)

    paginates = News.query.filter(*filters, News.status == 0).order_by(News.create_time.desc()).paginate(page, per_page,
                                                                                                         False)
    # 获取查询后的数据
    news_models = paginates.items

    # 获取分页的总页码
    total_page = paginates.pages

    # 获取当前页码
    current_page = paginates.page
    # 将内容转为字典，存入列表，传入模板
    news_list = []
    for news in news_models if news_models else []:
        news_list.append(news.to_dict())
    data = {
        "news_list": news_list,
        "total_page": total_page,
        "current_page": current_page
    }
    return jsonify(errno=RET.OK, errmsg='成功', data=data)


# 对模板进行渲染的一些显示的实现
@index_blue.route('/')
@user_login_data
def index():
    user = g.user
    # 二. 新闻分类展示
    # 从数据库获取分类数据
    try:
        categorys = Category.query.all()
    except Exception as e:
        logging.error(e)
    # 将模型转换为字典，存入列表中
    category_list = []
    for category in categorys if categorys else []:
        category_list.append(category.to_dict())

    # 三. 点击排行显示
    # 从数据库按照点击量降序查询前10条新闻
    try:
        news_click_models = News.query.order_by(News.clicks.desc()).limit(10)
    except Exception as e:
        logging.error(e)
    # 将模型转换为字典，存入列表中
    click_list = []
    for news in news_click_models if news_click_models else []:
        click_list.append(news.to_basic_dict())
    data = {
        "user": user.to_dict() if user else None,
        "category_list": category_list,
        "click_list": click_list
    }

    return render_template('news/index.html', data=data)


@index_blue.route('/favicon.ico')
def favicon():
    return current_app.send_static_file('news/favicon.ico')
